package com.joolun.weixin.config;

import cn.binarywang.wx.miniapp.api.WxMaService;
import cn.binarywang.wx.miniapp.api.impl.WxMaServiceImpl;
import cn.binarywang.wx.miniapp.bean.WxMaKefuMessage;
import cn.binarywang.wx.miniapp.bean.WxMaSubscribeMessage;
import cn.binarywang.wx.miniapp.config.impl.WxMaDefaultConfigImpl;
import cn.binarywang.wx.miniapp.message.WxMaMessageHandler;
import cn.binarywang.wx.miniapp.message.WxMaMessageRouter;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import lombok.extern.slf4j.Slf4j;
import me.chanjar.weixin.common.bean.result.WxMediaUploadResult;
import me.chanjar.weixin.common.error.WxErrorException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.annotation.Configuration;
import javax.annotation.PostConstruct;
import java.io.File;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * @author <a href="https://github.com/binarywang">Binary Wang</a>
 */
@Slf4j
@Configuration
@EnableConfigurationProperties(WxMaProperties.class)
public class WxMaConfiguration {
	private final WxMaProperties properties;

	private static final Map<String, WxMaMessageRouter> routers = Maps.newHashMap();
	private static Map<String, WxMaService> maServices;

	@Autowired
	public WxMaConfiguration(WxMaProperties properties) {
		this.properties = properties;
	}

	public static WxMaService getMaService(String appId) {
		WxMaService wxService = maServices.get(appId);
		if (wxService == null) {
			throw new IllegalArgumentException(String.format("未找到对应appId=[%s]的配置，请核实！", appId));
		}

		return wxService;
	}

	public static WxMaMessageRouter getRouter(String appId) {
		return routers.get(appId);
	}

	@PostConstruct
	public void init() {
		List<WxMaProperties.Config> configs = this.properties.getConfigs();
		if (configs == null) {
			throw new RuntimeException("大哥，拜托先看下项目首页的说明（readme文件），添加下相关配置，注意别配错了！");
		}

		maServices = configs.stream()
				.map(a -> {
					WxMaDefaultConfigImpl config = new WxMaDefaultConfigImpl();
					// 使用上面的配置时，需要同时引入jedis-lock的依赖，否则会报类无法找到的异常
					config.setAppid(a.getAppId());
					config.setSecret(a.getSecret());
					config.setToken(a.getToken());
					config.setAesKey(a.getAesKey());
					config.setMsgDataFormat(a.getMsgDataFormat());

					WxMaService service = new WxMaServiceImpl();
					service.setWxMaConfig(config);
					routers.put(a.getAppId(), this.newRouter(service));
					return service;
				}).collect(Collectors.toMap(s -> s.getWxMaConfig().getAppid(), a -> a));
	}

	private WxMaMessageRouter newRouter(WxMaService service) {
		final WxMaMessageRouter router = new WxMaMessageRouter(service);
		router
				.rule().handler(logHandler).next()
				.rule().async(false).content("订阅消息").handler(subscribeMsgHandler).end()
				.rule().async(false).content("文本").handler(textHandler).end()
				.rule().async(false).content("图片").handler(picHandler).end()
				.rule().async(false).content("二维码").handler(qrcodeHandler).end();
		return router;
	}

	private final WxMaMessageHandler subscribeMsgHandler = (wxMessage, context, service, sessionManager) -> {
		service.getMsgService().sendSubscribeMsg(WxMaSubscribeMessage.builder()
				.templateId("此处更换为自己的模板id")
				.data(Lists.newArrayList(
						new WxMaSubscribeMessage.Data("keyword1", "339208499")))
				.toUser(wxMessage.getFromUser())
				.build());
		return null;
	};

	private final WxMaMessageHandler logHandler = (wxMessage, context, service, sessionManager) -> {
		log.info("收到消息：" + wxMessage.toString());
		service.getMsgService().sendKefuMsg(WxMaKefuMessage.newTextBuilder().content("收到信息为：" + wxMessage.toJson())
				.toUser(wxMessage.getFromUser()).build());
		return null;
	};

	private final WxMaMessageHandler textHandler = (wxMessage, context, service, sessionManager) -> {
		service.getMsgService().sendKefuMsg(WxMaKefuMessage.newTextBuilder().content("回复文本消息")
				.toUser(wxMessage.getFromUser()).build());
		return null;
	};

	private final WxMaMessageHandler picHandler = (wxMessage, context, service, sessionManager) -> {
		try {
			WxMediaUploadResult uploadResult = service.getMediaService()
					.uploadMedia("image", "png",
							ClassLoader.getSystemResourceAsStream("tmp.png"));
			service.getMsgService().sendKefuMsg(
					WxMaKefuMessage
							.newImageBuilder()
							.mediaId(uploadResult.getMediaId())
							.toUser(wxMessage.getFromUser())
							.build());
		} catch (WxErrorException e) {
			e.printStackTrace();
		}

		return null;
	};

	private final WxMaMessageHandler qrcodeHandler = (wxMessage, context, service, sessionManager) -> {
		try {
			final File file = service.getQrcodeService().createQrcode("123", 430);
			WxMediaUploadResult uploadResult = service.getMediaService().uploadMedia("image", file);
			service.getMsgService().sendKefuMsg(
					WxMaKefuMessage
							.newImageBuilder()
							.mediaId(uploadResult.getMediaId())
							.toUser(wxMessage.getFromUser())
							.build());
		} catch (WxErrorException e) {
			e.printStackTrace();
		}

		return null;
	};

}
